package com.zzyl.nursing.service.impl;

import cn.hutool.core.collection.CollUtil;
import cn.hutool.core.date.LocalDateTimeUtil;
import cn.hutool.core.util.ObjectUtil;
import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
import com.github.pagehelper.PageHelper;
import com.github.pagehelper.PageInfo;
import com.zzyl.common.constant.HttpStatus;
import com.zzyl.common.core.domain.entity.SysUser;
import com.zzyl.common.core.page.TableDataInfo;
import com.zzyl.common.exception.base.BaseException;
import com.zzyl.common.utils.SecurityUtils;
import com.zzyl.common.utils.StringUtils;
import com.zzyl.nursing.domain.*;
import com.zzyl.nursing.dto.NursingTaskDto;
import com.zzyl.nursing.dto.TaskDto;
import com.zzyl.nursing.mapper.NursingTaskMapper;
import com.zzyl.nursing.service.*;
import com.zzyl.nursing.vo.NursingTaskVo;
import com.zzyl.serve.domain.NursingLevel;
import com.zzyl.serve.domain.NursingProject;
import com.zzyl.serve.service.INursingLevelService;
import com.zzyl.serve.service.INursingPlanService;
import com.zzyl.serve.service.INursingProjectService;
import com.zzyl.serve.vo.NursingPlanVo;
import com.zzyl.serve.vo.NursingProjectPlanVo;
import com.zzyl.system.service.ISysUserService;
import org.springframework.beans.BeanUtils;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Service;
import org.springframework.transaction.annotation.Transactional;

import java.time.LocalDateTime;
import java.time.LocalTime;
import java.time.temporal.ChronoUnit;
import java.util.ArrayList;
import java.util.Arrays;
import java.util.List;
import java.util.Map;
import java.util.stream.Collectors;

/**
 * 护理任务Service业务层处理
 * 
 * @author zsn
 * @date 2025-06-02
 */
@Service
public class NursingTaskServiceImpl extends ServiceImpl<NursingTaskMapper, NursingTask> implements INursingTaskService
{
    @Autowired
    private NursingTaskMapper nursingTaskMapper;

    /**
     * 查询护理任务
     * 
     * @param id 护理任务主键
     * @return 护理任务
     */
//    @Override
//    public NursingTask selectNursingTaskById(Long id)
//    {
//        return getById(id);
//    }


    @Autowired
    private IElderService elderService;
    @Autowired
    private INursingProjectService nursingProjectService;
    @Autowired
    private ISysUserService sysUserService;

    /**
     * 获取任务列表分页
     *
     * @param dto
     * @return
     */
    @Override
    public TableDataInfo<NursingTaskVo> list(TaskDto dto) {
        int pageNum = dto.getPageNum();
        int pageSize = dto.getPageSize();

        PageHelper.startPage(pageNum, pageSize);
        List<NursingTask> tasks = nursingTaskMapper.selectNursingTaskList(dto);
        //2  根据字符串类型的id,查询name,转换为VO
        List<NursingTaskVo> nursingTaskVos = new ArrayList<>();
        for (NursingTask task : tasks) {
            NursingTaskVo nursingTaskVo = new NursingTaskVo();
            nursingTaskVo.setNursingName(new ArrayList<>());
            BeanUtils.copyProperties(task, nursingTaskVo);
            // 2.1 根据id 查询护理员
            String nursingId = task.getNursingId();
            // 将字符串类型id转为 Long数组
            String[] nursingIds = nursingId.split(",");
            for (String id : nursingIds) {
                SysUser sysUser = sysUserService.selectUserById(Long.valueOf(id));
                nursingTaskVo.getNursingName().add(sysUser.getUserName());
            }
            nursingTaskVos.add(nursingTaskVo);
        }

        // 使用 PageInfo 获取 total
        PageInfo<NursingTask> pageInfo = new PageInfo<>(tasks);
        long total = pageInfo.getTotal();

        TableDataInfo<NursingTaskVo> tableDataInfo = new TableDataInfo<>();
        tableDataInfo.setTotal(total);
        tableDataInfo.setRows(nursingTaskVos);
        tableDataInfo.setCode(HttpStatus.SUCCESS);
        tableDataInfo.setMsg("查询成功");

        return tableDataInfo;
    }

    /**
     * 查询护理任务
     *
     * @param id 护理任务主键
     * @return 护理任务
     */
    @Autowired
    private ICheckInService checkInService;
    @Override
    public NursingTaskVo selectNursingTaskById(Long id) {

        NursingTask task = getById(id);
        NursingTaskVo vo = new NursingTaskVo();
        BeanUtils.copyProperties(task, vo);
        vo.setNursingName(new ArrayList<>());
        //1 根据字符串类型的id,查询name
        String[] ids = task.getNursingId().split(",");
        for (String nurseId : ids) {
            SysUser sysUser = sysUserService.selectUserById(Long.valueOf(nurseId));
            vo.getNursingName().add(sysUser.getUserName());
        }
        //2 根据老人id,查询年龄
        Elder elder = elderService.selectElderById(task.getElderId());
        vo.setAge(elder.getAge());
        //3 根据 项目id,查询项目名
        NursingProject project = nursingProjectService.selectNursingProjectById(task.getProjectId());
        vo.setProjectName(project.getName());

        //4 根据elder Id 查询老人,查询到老人对应的入住信息中的护理等级
        CheckIn checkIn = checkInService.lambdaQuery().eq(CheckIn::getElderId, task.getElderId()).one();
        if (checkIn != null) {
            vo.setNursingLevelName(checkIn.getNursingLevelName());
        }
        //5 根据 updateBy ,查询update
        if (task.getUpdateBy() != null) {
            SysUser sysUser = sysUserService.selectUserById(Long.valueOf(task.getUpdateBy()));
            vo.setUpdater(sysUser.getUserName());
        }
        return vo;
    }

    /**
     * 查询护理任务列表
     * 
     * @param nursingTask 护理任务
     * @return 护理任务
     */
//    @Override
//    public List<NursingTask> selectNursingTaskList(NursingTask nursingTask)
//    {
//        return nursingTaskMapper.selectNursingTaskList(nursingTask);
//    }

    /**
     * 新增护理任务
     * 
     * @param nursingTask 护理任务
     * @return 结果
     */
    @Override
    public int insertNursingTask(NursingTask nursingTask)
    {
        return save(nursingTask) ? 1 : 0;
    }

    /**
     * 修改护理任务
     * 
     * @param nursingTask 护理任务
     * @return 结果
     */
    @Override
    public int updateNursingTask(NursingTask nursingTask)
    {
        return updateById(nursingTask) ? 1 : 0;
    }

    /**
     * 批量删除护理任务
     * 
     * @param ids 需要删除的护理任务主键
     * @return 结果
     */
    @Override
    public int deleteNursingTaskByIds(Long[] ids)
    {
        return removeByIds(Arrays.asList(ids)) ? 1 : 0;
    }

    /**
     * 删除护理任务信息
     * 
     * @param id 护理任务主键
     * @return 结果
     */
    @Override
    public int deleteNursingTaskById(Long id)
    {
        return removeById(id) ? 1 : 0;
    }


    /**
     * 取消任务
     */
    @Override
    @Transactional(rollbackFor = Exception.class)
    public void cancelTask(NursingTaskDto dto) {
        //1.判断任务是否存在
        Long count = this.lambdaQuery().eq(NursingTask::getId, dto.getTaskId()).count();
        if (count<=0){
            throw new BaseException("任务不存在,无需取消");
        }
        //2.存在的时候取消,使用mp
        NursingTask nursingTask = new NursingTask();
        nursingTask.setId(dto.getTaskId());
        nursingTask.setCancelReason(dto.getReason());
        nursingTask.setUpdateTime(LocalDateTime.now());
        nursingTask.setStatus(3);
        //获取更新人id
        nursingTask.setUpdateBy(SecurityUtils.getLoginUser().getUserId().toString());
        //使用MPservice中的更新方法
        boolean result = this.updateById(nursingTask);
        if (!result){
            throw new BaseException("取消任务失败");
        }
    }

    /**
     * 任务执行
     *
     * @param
     */
    @Transactional(rollbackFor = Exception.class)
    @Override
    public void doTask(NursingTaskDto dto) {
        //1.判断护理任务id对应的护理任务是否存在
        Long count = this.lambdaQuery().eq(NursingTask::getId, dto.getTaskId()).count();
        if (count==0){
            throw new BaseException("任护任务表中没有对应的护理任务");
        }
        //2.进行任务执行
        NursingTask nursingTask = new NursingTask();
        BeanUtils.copyProperties(dto,nursingTask);

        //2.1补全基础属性
        nursingTask.setStatus(2);
        nursingTask.setUpdateTime(LocalDateTime.now());
        nursingTask.setUpdateBy(SecurityUtils.getLoginUser().getUserId().toString());
        nursingTask.setId(dto.getTaskId());
        boolean result = this.updateById(nursingTask);
        if (!result){
            throw new BaseException("任务执行失败");
        }
    }


    /**
     * 任务改期
     *
     * @param
     */
    @Override
    @Transactional(rollbackFor = Exception.class)
    public void updateTimeTask(NursingTaskDto dto) {
        //1.判断护理任务是否存在
        Long count = this.lambdaQuery().eq(NursingTask::getId, dto.getTaskId()).count();
        if (count==0){
            throw new BaseException("任护任务表中没有对应的护理任务");
        }
        NursingTask nursingTask = new NursingTask();
        BeanUtils.copyProperties(dto,nursingTask);
        //2.进行任务改期
        nursingTask.setUpdateTime(LocalDateTime.now());
        nursingTask.setUpdateBy(SecurityUtils.getLoginUser().getUserId().toString());
        nursingTask.setId(dto.getTaskId());
        boolean result = this.updateById(nursingTask);
        if (!result){
            throw new BaseException("任务执行失败");
        }
    }

    /**
     * 生成护理任务
     */

    @Autowired
    private ICheckInConfigService checkInConfigService;
    @Autowired
    private INursingLevelService nursingLevelService;
    @Autowired
    private INursingPlanService nursingPlanService;
    @Autowired
    private INursingElderService nursingElderService;


    /**
     * 生成月度护理任务
     *
     * @param elder 老人信息
     */
    @Override
    public void generateMonthlyTask(Elder elder) {
        // 基础校验1：校验
        if (elder == null || elder.getId() == null) {
            log.error("老人不能为空");
            return;
        }

        // 基础校验2：获取入住信息
        CheckIn checkIn = checkInService.lambdaQuery().eq(CheckIn::getElderId, elder.getId()).one();
        if (ObjectUtil.isEmpty(checkIn)) {
            log.error("入住信息不能为空");
            return;
        }

        // 基础校验3：获取入住及入住配置信息
        CheckInConfig checkInConfig = checkInConfigService.lambdaQuery().eq(CheckInConfig::getCheckInId, checkIn.getId()).one();
        if (ObjectUtil.isEmpty(checkInConfig)) {
            log.error("入住配置不能为空");
            return;
        }

        // 基础校验4：护理等级
        NursingLevel nursingLevel = nursingLevelService.selectNursingLevelById(checkInConfig.getNursingLevelId());
        if (ObjectUtil.isEmpty(nursingLevel)) {
            log.error("护理等级不能为空");
            return;
        }

        // 基础校验5：护理计划
        NursingPlanVo nursingPlanVo = nursingPlanService.selectNursingPlanById(nursingLevel.getLplanId());
        if (ObjectUtil.isEmpty(nursingPlanVo)) {
            log.error("护理计划不能为空");
            return;
        }

        // 基础校验6：护理计划项目计划列表
        List<NursingProjectPlanVo> projectPlans = nursingPlanVo.getProjectPlans();
        if(CollUtil.isEmpty(projectPlans)) {
            log.error("护理计划项目计划列表不能为空");
            return;
        }

        // 基础校验7：查询老人对应的护理员列表
        List<NursingElder> nursingElderList = nursingElderService.lambdaQuery().eq(NursingElder::getElderId, elder.getId()).list();
        if(CollUtil.isEmpty(nursingElderList)) {
            log.error("护理人员不能为空");
            return;
        }

        List<Long> list = nursingElderList.stream().map(NursingElder::getNursingId).collect(Collectors.toList());
        // 多个id用逗号分隔，转换为字符串
        String nursingIds = StringUtils.join(list, ",");

        // 查询所有的护理项目
        List<NursingProject> projectList = nursingProjectService.list();
        // 转换为map key是id,value是name
        Map<Long, String> nursingProjectMap = projectList.stream().collect(Collectors.toMap(NursingProject::getId, NursingProject::getName));

        // 开始时间
        LocalDateTime finalStartTime = LocalDateTime.now();

        // 组装护理任务
        List<NursingTask> nursingTasks = new ArrayList<>();
        projectPlans.forEach(planVo -> {
            // 执行频次
            Integer executeFrequency = Integer.valueOf(String.valueOf(planVo.getExecuteFrequency()));
            // 执行时间
            String executeTime = planVo.getExecuteTime();
            LocalTime localTime = LocalTime.parse(executeTime);
            // 开始执行时间
            LocalDateTime firstExecutionTime = LocalDateTime.of(finalStartTime.toLocalDate(), localTime);
            // 计算相差天数
            LocalDateTime monthEndTime = LocalDateTime.of(finalStartTime.getYear(), checkInConfig.getFeeStartDate().getMonth(), finalStartTime.toLocalDate().lengthOfMonth(), 23, 59);
            // 间隔天数
            //long diffDays = monthEndTime.toLocalDate().toEpochDay() - finalStartTime.toLocalDate().toEpochDay() + 1;
            long diffDays = Math.abs(LocalDateTimeUtil.between(finalStartTime, monthEndTime,  ChronoUnit.DAYS));
            if ("0".equals(planVo.getExecuteCycle())) {
                // 日
                generateTaskByDay(firstExecutionTime, diffDays, nursingTasks, executeFrequency, elder, planVo, nursingIds, nursingProjectMap);
            } else if ("1".equals(planVo.getExecuteCycle())) {
                // 周
                generateTaskByWeek(firstExecutionTime, diffDays, nursingTasks, executeFrequency, elder, planVo, monthEndTime, nursingIds, nursingProjectMap);
            } else {
                // 月
                generateTaskByMonth(firstExecutionTime, monthEndTime, nursingTasks, executeFrequency, elder, planVo, nursingIds, nursingProjectMap);
            }
        });

        if (CollUtil.isEmpty(nursingTasks)) {
            return;
        }
        this.saveBatch(nursingTasks);
    }


    /**
     * 按月创建任务
     *
     * @param firstExecutionTime
     * @param monthEndTime
     * @param nursingTasks
     * @param executeFrequency
     * @param elder
     * @param v
     */
    private void generateTaskByMonth(LocalDateTime firstExecutionTime, LocalDateTime monthEndTime, List<NursingTask> nursingTasks, Integer executeFrequency, Elder elder, NursingProjectPlanVo v, String nursingIds, Map<Long, String> nursingProjectMap) {
        LocalDateTime executionTime = firstExecutionTime;
        Integer diffDay = (monthEndTime.plusSeconds(1).getDayOfMonth() - executionTime.getDayOfMonth()) / executeFrequency;
        for (int x = 0; x < executeFrequency; x++) {
            // 根据时间差和执行顺序计算每个任务的具体时间
            LocalDateTime seconds = executionTime.plusDays(diffDay * x);
            // 初始化护理任务对象
            NursingTask nursingTask = getNursingTask(elder, v, nursingIds, seconds, nursingProjectMap);
            // 将生成的任务添加到任务列表中
            nursingTasks.add(nursingTask);
        }
    }

    /**
     * 根据周为单位创建护理任务
     *
     * 此方法旨在根据给定的开始时间、差异天数、执行频率等参数，为指定的老人生成护理任务列表
     * 它考虑了跨年情况以及每月结束时间的限制，避免生成越过月末的有效任务
     *
     * @param firstExecutionTime 首次执行时间，用于计算后续任务的时间点
     * @param diffDays 任务之间的时间差，以天为单位，此方法专注于周级别，故此值应与7的倍数有关
     * @param nursingTasks 护理任务列表，方法将新生成的任务添加到此列表中
     * @param executeFrequency 执行频率，决定每周内任务执行的次数
     * @param elder 老人信息对象，任务相关的老人信息由此对象提供
     * @param v 护理项目计划的视图对象，包含项目ID等信息
     * @param monthEndTime 每月结束时间的限制，确保任务不会跨月生成
     * @param nursingIds 护理员ID列表，以字符串形式传递，用于分配任务给护理员
     */
    private void generateTaskByWeek(LocalDateTime firstExecutionTime, long diffDays, List<NursingTask> nursingTasks, Integer executeFrequency, Elder elder, NursingProjectPlanVo v, LocalDateTime monthEndTime, String nursingIds, Map<Long, String> nursingProjectMap) {
        int i;
        // 以7天为步长遍历差异天数，创建每周的任务
        for (i = 0; i < diffDays - 7; i = i + 7) {
            // 计算每周结束时间
            LocalDateTime dayEndTime = LocalDateTime.of(firstExecutionTime.plusDays(i + 7).toLocalDate(), LocalTime.of(23, 59));
            // 计算本周的执行起始时间
            LocalDateTime executionTime = firstExecutionTime.plusDays(i);
            // 根据执行频率计算时间差，用于确定本周内各任务的时间点
            Integer diffDay = (dayEndTime.plusSeconds(1).getDayOfYear() - executionTime.getDayOfYear()) / executeFrequency;
            // 根据执行频率生成本周的任务
            for (int x = 0; x < executeFrequency; x++) {
                // 根据时间差和执行顺序计算每个任务的具体时间
                LocalDateTime seconds = executionTime.plusDays(diffDay * x);
                // 初始化护理任务对象
                NursingTask nursingTask = getNursingTask(elder, v, nursingIds, seconds,nursingProjectMap);
                // 将生成的任务添加到任务列表中
                nursingTasks.add(nursingTask);
            }
        }

        // 处理边界情况，当i超过diffDays-7且小于diffDays时
        if (i > diffDays - 7 && i < diffDays) {
            // 计算到达diffDays时的结束时间
            LocalDateTime dayEndTime = LocalDateTime.of(firstExecutionTime.plusDays(i + 7).toLocalDate(), LocalTime.of(23, 59));
            // 如果结束时间与开始时间年份不同，则返回，避免跨年任务
            if (ObjectUtil.notEqual(dayEndTime.getYear(), firstExecutionTime.getYear())) {
                return;
            }
            // 计算执行时间
            LocalDateTime executionTime = firstExecutionTime.plusDays(i);
            // 计算时间差，确定任务时间点
            Integer diffDay = (dayEndTime.plusSeconds(1).getDayOfYear() - executionTime.getDayOfYear()) / executeFrequency;
            // 根据执行频率生成任务，直到月末
            for (int x = 0; x < executeFrequency; x++) {
                // 计算每个任务的时间
                LocalDateTime seconds = executionTime.plusDays(diffDay * x);
                // 如果任务时间超过月末结束时间，则停止生成
                if (seconds.isAfter(monthEndTime)) {
                    break;
                }
                // 初始化护理任务对象
                NursingTask nursingTask = getNursingTask(elder, v, nursingIds, seconds,nursingProjectMap);
                // 将生成的任务添加到任务列表中
                nursingTasks.add(nursingTask);
            }
        }

    }

/**
 * 按日创建任务
 * @param firstExecutionTime
 * @param diffDays
 * @param nursingTasks
 * @param executeFrequency
 * @param elder
 * @param v
 */
    /**
     * 根据天数创建护理任务
     * <p>
     * 该方法根据首次执行时间、任务间隔天数、护理任务列表、执行频率、老人信息和护理项目计划数据，
     * 创建一系列的护理任务。任务的创建以天为单位，直到达到任务间隔天数为止。
     *
     * @param firstExecutionTime 首次执行时间，标志着任务创建的起始时间
     * @param diffDays           任务间隔天数，确定需要创建任务的天数范围
     * @param nursingTasks       护理任务列表，创建的任务将被添加到此列表中
     * @param executeFrequency   执行频率，决定一天内任务执行的次数
     * @param elder              老人对象，包含执行任务的老人的信息
     * @param v                  护理项目计划数据对象，包含护理项目的相关信息
     */
    private void generateTaskByDay(LocalDateTime firstExecutionTime, long diffDays, List<NursingTask> nursingTasks, Integer executeFrequency, Elder elder, NursingProjectPlanVo v, String nursingIds, Map<Long, String> nursingProjectMap) {
        // 遍历每一天，从首次执行时间开始，直到达到任务间隔天数
        for (int i = 0; i < diffDays; i++) {
            // 计算每一天的任务执行起始时间
            LocalDateTime executionTime = firstExecutionTime.plusDays(i);
            // 确定当天任务执行的结束时间（当天的23:59）
            LocalDateTime dayEndTime = LocalDateTime.of(executionTime.toLocalDate(), LocalTime.of(23, 59));
            // 计算当天内每次执行任务之间的小时差，用于确定执行时间点
            Integer diffHour = (dayEndTime.plusSeconds(1).getHour() - executionTime.getHour()) / executeFrequency;
            // 根据执行频率创建任务
            for (int x = 0; x < executeFrequency; x++) {
                // 计算每个任务的具体执行时间
                LocalDateTime seconds = executionTime.plusHours(diffHour * x);
                // 创建新的护理任务对象
                NursingTask nursingTask = getNursingTask(elder, v, nursingIds, seconds,nursingProjectMap);
                // 将生成的任务添加到任务列表中
                nursingTasks.add(nursingTask);
            }
        }
    }

    /**
     * 获取护理任务对象
     * @param elder
     * @param v
     * @param nursingIds
     * @param seconds
     * @return
     */
    private NursingTask getNursingTask(Elder elder, NursingProjectPlanVo v, String nursingIds, LocalDateTime seconds, Map<Long, String> nursingProjectMap) {
        NursingTask nursingTask = new NursingTask();
        // 设置任务状态
        nursingTask.setStatus(1);
        // 设置护理员ID列表
        nursingTask.setNursingId(nursingIds);
        // 老人姓名
        nursingTask.setElderName(elder.getName());
        // 设置床位号
        nursingTask.setBedNumber(elder.getBedNumber());
        // 设置预估服务时间
        nursingTask.setEstimatedServerTime(seconds);
        // 设置项目ID
        nursingTask.setProjectId(Long.valueOf(v.getProjectId()));
        // 设置老人ID
        nursingTask.setElderId(elder.getId());
        // 匹配护理项目
        nursingTask.setProjectName(nursingProjectMap.get(Long.valueOf(nursingTask.getProjectId())));
        // 设置任务类型
        nursingTask.setTaskType(2);
        return nursingTask;
    }
}
